package at.ac.univie.ke.neuranetwork;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.neuroph.core.NeuralNetwork;
import org.neuroph.nnet.MultiLayerPerceptron;

public class NetworkWriter {
	public static int INPUT_NEURONS=784;

	public static int HIDDEN_NEURONS=30;

	public static int OUTPUT_NEURONS=10;

	public static void main(String[] args) throws IOException {
		
		List<Integer> layers = new ArrayList<Integer>();
		layers.add(INPUT_NEURONS);
		layers.add(HIDDEN_NEURONS);
		layers.add(OUTPUT_NEURONS	);
	
		// create new perceptron network
		NeuralNetwork neuralNetwork = new MultiLayerPerceptron(layers);
		
		/*Neural Netwokr speichern */
		neuralNetwork.save("/home/daniel/workspace/ppmfiles/"+INPUT_NEURONS + "i-" +HIDDEN_NEURONS + "h-" +OUTPUT_NEURONS + "o.nnet" );

		System.out.println("file saved.");
	}

}